/*
 * SoapUI, Copyright (C) 2004-2017 SmartBear Software
 *
 * Licensed under the EUPL, Version 1.1 or - as soon as they will be approved by the European Commission - subsequent 
 * versions of the EUPL (the "Licence"); 
 * You may not use this work except in compliance with the Licence. 
 * You may obtain a copy of the Licence at: 
 * 
 * http://ec.europa.eu/idabc/eupl 
 * 
 * Unless required by applicable law or agreed to in writing, software distributed under the Licence is 
 * distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either 
 * express or implied. See the Licence for the specific language governing permissions and limitations 
 * under the Licence. 
 */

package com.eviware.soapui.impl.wsdl.submit.transports.http.support.attachments;

import com.eviware.soapui.SoapUI;
import com.eviware.soapui.impl.wsdl.WsdlRequest;
import com.eviware.soapui.impl.wsdl.support.soap.SoapVersion;
import org.apache.http.Header;
import org.apache.http.entity.AbstractHttpEntity;
import org.apache.http.message.BasicHeader;

import javax.mail.MessagingException;
import javax.mail.internet.MimeMessage;
import javax.mail.internet.MimeMultipart;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

/**
 * MimeMessage request class
 *
 * @author ole.matzura
 */

public class WsdlRequestMimeMessageRequestEntity extends AbstractHttpEntity {
    private final MimeMessage message;
    private final boolean isXOP;
    private final WsdlRequest wsdlRequest;

    public WsdlRequestMimeMessageRequestEntity(MimeMessage message, boolean isXOP, WsdlRequest wsdlRequest) {
        this.message = message;
        this.isXOP = isXOP;
        this.wsdlRequest = wsdlRequest;
    }

    public long getContentLength() {
        try {
            DummyOutputStream out = new DummyOutputStream();
            writeTo(out);
            return out.getSize();
        } catch (Exception e) {
            SoapUI.logError(e);
            return -1;
        }
    }

    public Header getContentType() {
        try {
            SoapVersion soapVersion = wsdlRequest.getOperation().getInterface().getSoapVersion();

            if (isXOP) {
                String header = message.getHeader("Content-Type")[0];

                return new BasicHeader("Content-Type", AttachmentUtils.buildMTOMContentType(header,
                        wsdlRequest.getAction(), soapVersion));
            } else {
                String header = message.getHeader("Content-Type")[0];
                int ix = header.indexOf("boundary");

                return new BasicHeader("Content-Type", "multipart/related; type=\"" + soapVersion.getContentType()
                        + "\"; " + "start=\"" + AttachmentUtils.ROOTPART_SOAPUI_ORG + "\"; " + header.substring(ix));
            }
        } catch (MessagingException e) {
            SoapUI.logError(e);
        }

        return null;
    }

    @Override
    public boolean isRepeatable() {
        return true;
    }

    @Override
    public void writeTo(OutputStream arg0) throws IOException {
        try {
            arg0.write("\r\n".getBytes());
            ((MimeMultipart) message.getContent()).writeTo(arg0);
        } catch (Exception e) {
            SoapUI.logError(e);
        }
    }

    @Override
    public InputStream getContent() throws IOException {
        try {
            return message.getInputStream();
        } catch (MessagingException e) {
            throw new IOException(e);
        }
    }

    @Override
    public boolean isStreaming() {
        return false;
    }

    public static class DummyOutputStream extends OutputStream {
        private int intLength;
        private long size = 0;

        public DummyOutputStream() {
            ByteArrayOutputStream tempOut = new ByteArrayOutputStream();
            tempOut.write(1);
            intLength = tempOut.toByteArray().length;
        }

        @Override
        public void write(int b) throws IOException {
            size += intLength;
        }

        public long getSize() {
            return size;
        }
    }
}
